avatar

Catalog
剑指offer 重建二叉树

题目描述

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的J前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。

解题思路

前序遍历的顺序为根->左->右,所以前序遍历第一个数字是根节点,中序遍历的顺序为左->根->右,在数组中找到根节点可以分出左子树和右子树,所以

分割后的中序遍历

左子树为{4,7,2}右子树为{5,3,8,6}根节点为1

前序遍历

左子树为{2,4,7},右子树为{3,5,6,8},根节点为1

代码如下

java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
import java.util.Arrays;
public class Solution {
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
if(pre.length==0|in.length==0){//终止条件
return null;
}
TreeNode root = new TreeNode(pre[0]);
//在中序遍历中寻找根节点
for(int i =0;i<in.length;i++){
if(in[i]==pre[0]){
//分离左子树,递归
root.left=reConstructBinaryTree(Arrays.copyOfRange(pre,1,i+1),Arrays.copyOfRange(in,0,i));
// 分离右子树,递归
root.right=reConstructBinaryTree(Arrays.copyOfRange(pre,i+1,pre.length),Arrays.copyOfRange(in,i+1,in.length));
break;
}
}return root;
}
}

盲点

对java的类不熟悉,TreeNode和Arrays的方法不熟悉,加强学习

Author: Jacob
Link: http://yoursite.com/2019/09/16/%E5%89%91%E6%8C%87offer%20%E7%AC%AC4%E9%A2%98%20%E9%87%8D%E5%BB%BA%E4%BA%8C%E5%8F%89%E6%A0%91/
Copyright Notice: All articles in this blog are licensed under CC BY-NC-SA 4.0 unless stating additionally.
Donate
  • 微信
    微信
  • 支付寶
    支付寶